package com.tech.repository.dto;

import com.tech.repository.entity.Goods;
import com.tech.repository.entity.GoodsImage;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.awt.image.BufferedImage;
import java.util.List;

/**
 * 封装的商品及商品图片对象
 */
@Data
@NoArgsConstructor
@AllArgsConstructor
public class GoodsInfo {
	private Goods goods;
	private List<GoodsImage> images;

	public GoodsInfo(Goods goods) {
		this.goods = goods;
	}

	/**
	 * 获取商品的第一张图片
	 *
	 * @return 如果存在就返回第一张图片，否则返回null
	 */
	public BufferedImage getFirstImage() {
		return (images != null && images.size() != 0) ? images.get(0).getImage() : null;
	}
}
